package 中等.队列;

import java.util.PriorityQueue;

/**
 * 给你一个二维整数数组 orders ，其中每个 orders[i] = [pricei, amounti, orderTypei]
 * 表示有 amounti 笔类型为 orderTypei 、价格为 pricei 的订单。
 * 订单类型 orderTypei 可以分为两种：
 * 0 表示这是一批采购订单 buy
 * 1 表示这是一批销售订单 sell
 * 注意，orders[i] 表示一批共计 amounti 笔的独立订单，这些订单的价格和类型相同。
 * 对于所有有效的 i ，由 orders[i] 表示的所有订单提交时间均早于 orders[i+1] 表示的所有订单。
 * 存在由未执行订单组成的 积压订单 。积压订单最初是空的。提交订单时，会发生以下情况：
 * 如果该订单是一笔采购订单 buy ，则可以查看积压订单中价格 最低 的销售订单 sell 。
 * 如果该销售订单 sell 的价格 低于或等于 当前采购订单 buy 的价格，则匹配并执行这
 * 两笔订单，并将销售订单 sell 从积压订单中删除。否则，采购订单 buy 将会添加到积压订单中。
 * 反之亦然，如果该订单是一笔销售订单 sell ，则可以查看积压订单中价格 最高 的采购
 * 订单 buy 。如果该采购订单 buy 的价格 高于或等于 当前销售订单 sell 的价格，则
 * 匹配并执行这两笔订单，并将采购订单 buy 从积压订单中删除。否则，销售订单 sell 将会添加到积压订单中。
 * 输入所有订单后，返回积压订单中的 订单总数 。由于数字可能很大，所以需要返回对 109 + 7 取余的结果。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/number-of-orders-in-the-backlog
 */
public class 积压订单中的订单总数_1801 {

    public static void main(String[] args) {

        System.out.println(getNumberOfBacklogOrders(new int[][]{{7, 1000000000, 1}, {15, 3, 0}, {5, 999999995, 0}, {5, 1, 1}}));

    }

    /**
     * 优先队列模拟
     *
     * @param orders
     * @return
     */
    public static int getNumberOfBacklogOrders(int[][] orders) {
        PriorityQueue<int[]> sellQueue = new PriorityQueue<>((a, b) -> a[0] - b[0]);
        PriorityQueue<int[]> buyQueue = new PriorityQueue<>((a, b) -> b[0] - a[0]);

        for (int[] order : orders) {
            if (order[2] == 0) {
                while (!sellQueue.isEmpty() && sellQueue.peek()[0] <= order[0] &&
                        order[1] > 0) {
                    int[] sell = sellQueue.poll();
                    if (sell[1] >= order[1]) {
                        sell[1] -= order[1];
                        order[1] = 0;
                        if (sell[1] > 0) sellQueue.add(sell);
                    } else {
                        order[1] -= sell[1];
                    }
                }
                if (order[1] > 0) {
                    buyQueue.add(order);
                }
            } else {
                while (!buyQueue.isEmpty() && buyQueue.peek()[0] >= order[0] &&
                        order[1] > 0) {
                    int[] buy = buyQueue.poll();
                    if (buy[1] >= order[1]) {
                        buy[1] -= order[1];
                        order[1] = 0;
                        if (buy[1] > 0) buyQueue.add(buy);
                    } else {
                        order[1] -= buy[1];
                    }
                }
                if (order[1] > 0) {
                    sellQueue.add(order);
                }
            }
        }
        int count = 0;
        while (!sellQueue.isEmpty()) {
            count += sellQueue.poll()[1];
            count %= 1000000007;
        }
        while (!buyQueue.isEmpty()) {
            count += buyQueue.poll()[1];
            count %= 1000000007;
        }
        return count;
    }

}
